package 二叉搜索树中的插入操作;

/**
 * @author zhengTao
 * @version 1.0
 * @description: 二叉搜索树中的插入操作
 * 给定二叉搜索树（BST）的根节点root和要插入树中的值value，将值插入二叉搜索树。 返回插入后二叉搜索树的根节点。 输入数据 保证 ，新值和原始二叉搜索树中的任意节点值都不同。
 * 注意，可能存在多种有效的插入方式，只要树在插入后仍保持为二叉搜索树即可。 你可以返回 任意有效的结果 。
 * @date 2022/11/7 10:21
 */
public class Demo701 {


    //Definition for a binary tree node.
    public class TreeNode {
        int val;
        TreeNode left;
        TreeNode right;

        TreeNode() {
        }

        TreeNode(int val) {
            this.val = val;
        }

        TreeNode(int val, TreeNode left, TreeNode right) {
            this.val = val;
            this.left = left;
            this.right = right;
        }
    }

    //题目说的很难，但其实按照二叉搜素树的正常遍历顺序，可以不用改变树的结构，返回原始的root节点即可
    public TreeNode insertIntoBST(TreeNode root, int val) {
        if (root == null) {
            return new TreeNode(val);
        }
        //保存根节点，最后好返回
        TreeNode cur = root;
        //保存前一个节点
        TreeNode pre = root;
        //遍历
        while (root != null) {
            pre = root;
            if (root.val > val) {
                root=root.left;
            } else {
                root=root.right;
            }
        }
        if (pre.val > val) {
            pre.left = new TreeNode(val);
        } else {
            pre.right = new TreeNode(val);
        }
        return cur;
    }

}
